package com.base.util;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Writer;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;

public class FreeMarkerUtil {

	/**
	 * templatePath模板文件存放路径,templateName 模板文件名称,filename 生成的文件名称
	 * 
	 * @param templatePath
	 * @param templateName
	 * @param fileName
	 * @param root
	 */
	public static void analysisTemplate(String templatePath, String templateName, String fileName, Object root) {
		try {
			// 初使化FreeMarker配置
			Configuration config = new Configuration();
			// 设置要解析的模板所在的目录，并加载模板文件
			config.setDirectoryForTemplateLoading(new File(templatePath));
			// 设置包装器，并将对象包装为数据模型
			config.setObjectWrapper(new DefaultObjectWrapper());
			// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
			// 否则会出现乱码
			Template template = config.getTemplate(templateName, "UTF-8");
			// 合并数据模型与模板
			FileOutputStream fos = new FileOutputStream(fileName);
			Writer out = new OutputStreamWriter(fos, "UTF-8");
			template.process(root, out);
			out.flush();
			out.close();
		} catch (IOException e) {
			e.printStackTrace();
		} catch (TemplateException e) {
			e.printStackTrace();
		}
	}

	public static void analysisTemplate(File templateFile, File targetFile, Object root) {
		String templatePath = templateFile.getParent();
		String templateName = templateFile.getName();
		try {
			// 初使化FreeMarker配置
			Configuration config = new Configuration();
			// 设置要解析的模板所在的目录，并加载模板文件
			config.setDirectoryForTemplateLoading(new File(templatePath));
			// 设置包装器，并将对象包装为数据模型
			config.setObjectWrapper(new DefaultObjectWrapper());
			// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
			// 否则会出现乱码
			Template template = config.getTemplate(templateName, "UTF-8");
			// 合并数据模型与模板
			FileOutputStream fos = new FileOutputStream(targetFile);
			Writer out = new OutputStreamWriter(fos, "UTF-8");
			template.process(root, out);
			out.flush();
			out.close();
		} catch (IOException e) {
			e.printStackTrace();
		} catch (TemplateException e) {
			e.printStackTrace();
		}
	}

	public static void printByTmplate(File templateFile, Object root) {
		String templatePath = templateFile.getParent();
		String templateName = templateFile.getName();
		try {
			// 初使化FreeMarker配置
			Configuration config = new Configuration();
			// 设置要解析的模板所在的目录，并加载模板文件
			config.setDirectoryForTemplateLoading(new File(templatePath));
			// 设置包装器，并将对象包装为数据模型
			config.setObjectWrapper(new DefaultObjectWrapper());
			// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
			// 否则会出现乱码
			Template template = config.getTemplate(templateName, "UTF-8");
			// 合并数据模型与模板
			PrintWriter out = new PrintWriter(System.out);
			template.process(root, out);
		} catch (IOException e) {
			e.printStackTrace();
		} catch (TemplateException e) {
			e.printStackTrace();
		}
	}

	public static String stringByTemplate(File templateFile, Object root) {
		String templatePath = templateFile.getParent();
		String templateName = templateFile.getName();
		StringBuffer sb = new StringBuffer();
		try {
			// 初使化FreeMarker配置
			Configuration config = new Configuration();
			// 设置要解析的模板所在的目录，并加载模板文件
			config.setDirectoryForTemplateLoading(new File(templatePath));
			// 设置包装器，并将对象包装为数据模型
			config.setObjectWrapper(new DefaultObjectWrapper());
			// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
			// 否则会出现乱码
			Template template = config.getTemplate(templateName, "UTF-8");
			// 合并数据模型与模板
			StringWriter out = new StringWriter();
			template.process(root, out);
			sb.append(out.toString());
		} catch (IOException e) {
			e.printStackTrace();
		} catch (TemplateException e) {
			e.printStackTrace();
		}
		return sb.toString();
	}

	public static String stringByTemplate2(File templateFile, Object root) {
		String templatePath = templateFile.getParent();
		String templateName = templateFile.getName();
		StringBuffer sb = new StringBuffer();
		try {
			// 初使化FreeMarker配置
			Configuration config = new Configuration();
			// 设置要解析的模板所在的目录，并加载模板文件
			config.setDirectoryForTemplateLoading(new File(templatePath));
			// 设置包装器，并将对象包装为数据模型
			config.setObjectWrapper(new DefaultObjectWrapper());
			// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
			// 否则会出现乱码
			Template template = config.getTemplate(templateName, "UTF-8");
			sb.append(template.toString());
		} catch (IOException e) {
			e.printStackTrace();
		}
		return sb.toString();
	}

	public static String stringByStr(String ftlStr, Object root) throws IOException, TemplateException {
		StringBuffer sb = new StringBuffer();

		// 初使化FreeMarker配置
		Configuration config = new Configuration();
		// 设置要解析的模板所在的目录，并加载模板文件
		// config.setDirectoryForTemplateLoading(new File(templatePath));
		// 设置包装器，并将对象包装为数据模型
		config.setObjectWrapper(new DefaultObjectWrapper());
		// 获取模板,并设置编码方式，这个编码必须要与页面中的编码格式一致
		// 否则会出现乱码

		Template template = new Template(null, new StringReader(ftlStr), config);
		// 合并数据模型与模板
		StringWriter out = new StringWriter();
		template.process(root, out);

		sb.append(out.toString());

		return sb.toString();
	}

	public static void main(String[] args) {
		String templatePath = FreeMarkerUtil.class.getResource("/").getPath() + "template";
		System.out.println(templatePath);
		System.out.println(new File(templatePath).exists());

	}

}
